from django.http import HttpResponse

from django_redis import get_redis_connection

from rest_framework.generics import GenericAPIView
from rest_framework.response import Response
from rest_framework.views import APIView

from celery_tasks.sms.yuntongxun.sms import CCP
from celery_tasks.sms.tasks import send_sms_code

from meiduo_mall.libs.captcha.captcha import captcha

from verifications import constants

from . import serializers
import random


# Create your views here.
class ImageCodeView(APIView):
    """
    图片验证码
    http://127.0.0.1:8000/image_codes/abcd/
    api.meiduo.site:8000/image_codes/abcd/

    """

    def get(self, request, image_code_id):
        """
        提供图片验证码
        :param request:
        :param image_code_id:
        :return:
        """
        # 生成图片验证码 text是验证码内容 需要保存到Redis image时图片验证码数据 需要相应给前端
        text, image = captcha.generate_captcha()
        print("text:", text)

        # django-redis提供了get_redis_connection方法 来传递redis的配置名称可获取redis的连接对象
        # 通过redis连接对象可以执行redis命令
        # 将图片验证码内容存储到redis数据库中
        redis_conn = get_redis_connection("verify_codes")
        # redis数据库中 保存图片的名字 内容 和有效时间
        redis_conn.setex("img_%s" % image_code_id, constants.IMAGE_CODE_REDIS_EXPIRES, text)

        # 将图片验证码图片数据相应到前端
        return HttpResponse(image, content_type="images/jpg")


class SMSCodeView(GenericAPIView):
    """
    短信验证码
    mobile
    image_code_id
    text
    """
    # 指定序列化器
    serializer_class = serializers.ImageCodeCheckSerializer

    def get(self, request, mobile):
        # 创建序列化器对象 data=request.query_params 是将url中的查询字符串传入到序列器中
        serializer = self.get_serializer(data=request.query_params)

        # 开启校验
        serializer.is_valid(raise_exception=True)

        # 生成短信验证码 0
        sms_code = "%06d" % random.randint(0, 999999)
        print("sms_code:", sms_code)

        # 存储短信验证码到redis数据库
        redis_conn = get_redis_connection("verify_codes")
        # redis_conn.setex('sms_%s' % mobile, constants.SMS_CODE_REDIS_EXPIRES, sms_code)
        #
        # # 记录用户发送短信的频次
        # redis_conn.setex('send_flag_%s' % mobile, constants.SEND_SMS_CODE_INTERVAL, 1)

        # 创建管道
        pl = redis_conn.pipeline()
        pl.setex('sms_%s' % mobile, constants.SMS_CODE_REDIS_EXPIRES, sms_code)
        # 记录用户发送短信的频次
        pl.setex('send_flag_%s' % mobile, constants.SEND_SMS_CODE_INTERVAL, 1)
        # 开启执行
        pl.execute()

        # 发送短信验证码
        CCP().send_template_sms(mobile, [sms_code, constants.SMS_CODE_REDIS_EXPIRES // 60],
                                constants.SEND_SMS_TEMPLATE_ID)

        # 调用异步任务
        # ，将异步任务发送不到broker任务队列中
        # 使用delay函数
        send_sms_code.delay(mobile, sms_code)

        # 响应结果
        return Response({'message': 'OK'})
